# __init__.py
import os
import sys
from pathlib import Path

# 使用 pathlib
PROJECT_ROOT = Path(__file__).parent.parent.resolve()

# 使用传统方式，但更加健壮
PROJECT_ROOT_STR = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
if PROJECT_ROOT_STR not in sys.path:
    sys.path.insert(0, PROJECT_ROOT_STR)
    
# 确保路径存在的辅助函数
def ensure_dir_exists(path):
    """确保目录存在，不存在则创建"""
    Path(path).mkdir(parents=True, exist_ok=True)
    return path

# 常用子目录
PACKAGE_ROOT = Path(__file__).parent
DATA_DIR = PROJECT_ROOT / 'datasets'
MODELS_DIR = PROJECT_ROOT / 'models'